Updating Viz

Fixes #7 and #6