Updates to add docs, touching docstrings, etc