Finished Hw2 part1