Update with Migrations Code first

Update with Migrations Code first