Added db models with migration