main.py updates