Added hook to migrate in production