Could you please add a toy example as doctest in the docstring? I am having a hard time understanding the goal of this one.