Added to update Method