Minor cleanup
Added some more docstrings to functions that were lacking it.