Python File Reading and User Input - Assignment Solution