Second try at hw2 part1