Trap Water update

I copied somebody else's code  sigh...