Update Unit 1 Problem 21.py