Use docstrings instead of comments