Adding solution to problem_20