Update problem_020.py