Improve code format in docstrings