Correction on the docstrings and comments