Moving declarative_base to models.py