slight comment and docstring updates