finished problem 1 and unit test