fixed answer for assignment 1