Edit run.py for server