Update mini-project-2.py