Added model.py so we can use the same model module