Currently all models are in models.py It's getting messy, extract each entity type into their own file