Solved pset 2, problem 1 and test cases