problem_1.py : Completed