Finish problem 2 lol