Modifying schema slightly and updating run.py