Update for Waterflow Thresholding

Still need to figure out a thresholding argument, and possibly convert
towards Matplotlib instead of PIL for better access to numpy.