Minor docstring update and cleanup