Much better, but you'll probably want to update the docstring too.