Updated for flask-migrate