added solution to water_level problem