Added the docstrings and few not crucial changes.