Done with problem 3