More visualization changes