Code I'm using to visualize data