Update models.py and admin.py