Update save method