Make running main.py directly work better