Moved run loop back to model.py as this makes more sense