Redo model.py