Update to add Migrations and Notification callback