Done with problem 6