New visuals

fixed several issues.

Maybe fix this too?