models and migration

updated models file and migration file