Update models.py - use User for location