updating main.py to have main at end