Reorganise. Main in main.py.