update ex20.py