Started part 2 of assignment 3