perform watershed plane-by-plane instead of in 3D

I've added a new function to do what's described in the title. I've also taken the opportunity to update the doc on the watershed function and implement a kwarg that wasn't actually implemented