update project3 main.py