Update ex31.py