changed run.py
Revamped main() in run.py