Also, please add a test, and update the docstring?