New visuals

fixed several issues.

missing docstring