Fix vertex shader

Fixed vertex shader so colors look more normal.  Cleaned up the code
in the vertex shader to be easier to read.