Update mini-project-4.py