updated screenshot, refector docstring