Completed visualization, more or less. Should still fiddle with colors