Finish problem 4