Minor tweaks.
Code commenting/docstrings