Update models.py

Add name