Some visualization changes