Somewhat decent version.  Should prob work on the visualization