finished assignment 1 problem 5