Part 1 for Assign2 complete