I think this will work with the models.py file