Here is the complete code in a wdl file, because there are some troubles with copy and paste the code:

water_fx.wdl