problem_10.py : Completed