Add migration handler