Move model classes to models.py