code review on models.py