Add two models im models.py