Update models.py

Acho que assim fica mais performático e pratico.