Done with problem 5