Continued working on model.py