updated main.py