steps: minor refactor, whitespace, and docstrings