modify main.py for linux