Small changes on models.py