Update water flow: add animation