Add migration for updated disk model