Add Returns clause to docstring