Cleaned up example in docstring