comments added to model.py