Minor changes like docstrings and logging calls.