fix: update the sketch with the modified code