Replaced header comments with docstrings