More visualization tweaks