Docstrings and minor tweaks for cli runner