finish problem_five with tests