Update mini-project-5.py