Add Viz support (beta)