finishing up problem 5