MNT: Cleaned up imports in run.py