This is nitpicky, but could we maybe add a docstring to shape?