problem016.py : Completed