problem022.py : Completed