Added model to migration to follow best practices