Add modeles.py file