finish models in models.py