Added docstrings
Added more documentation