Update Problem_21.py