First working version of viz, still needs some tweaks