run.py works better now