Fixing run.py
Finally getting import structure working