Updated the main.py should