Minor comment and docstring improvements