Started problem 3, need to test