Finished problem_2