Finish problem 3. All tests pass