fixed some visualization issues