Don't forget to update docstrings