Fixes to element visualization

"color group" sort of starting to work