Ha! I fogot to update the docstring. Thanks.