Adding migration and model