minor changes to docstring, showing images during run