problem020.py : Completed