assignment 1, part 2 done