Add more docstrings - thanks @manishtomar