Update water_sample.rb