Moved main.py to top level