cleaned up some visualization stuff among other things