Updates to viz tools