Cleaned up issues with WaterBox subclass docstrings.