add part 2 solution to assignment 3