Adding JsonNode code for updating as well, this was initially missed