Add support for docstrings