Minor docstring and logging improvements