Assignment 2 end of part 1