Added water automaton class