Done with problem 2, started problem 3