Update models.py

changed context to JSONField