Moved declarative_base to models.py