Added methods in models.py