Improved few docstrings