Water balance modelling in a semi-arid environment