Some shader cleanup.
Nicer colors, arrows.