update ex21.py