More visualization prototype work