completed code for assignment 1 part 1