some visualization improvements