Update ex20.py