Edit: Add docstring