Remove models.py and move to __init__.py