Switched to sqlalchemy