Fixed main.py