Fix seed training data and update utility to use sqlalchemy