finished problem set 1 on hw2, with tests