add water class computation