Water system modelling