Create models.py

Created models to use it as a module