Adding move to next logic to water puzzle