finish problem_two_question_two along with tests