Can you add a docstring to this one too describing what its doing and what kind of data structure it returns?