kind of done with problem 2