Update assign9_4.py