Vertex shader now uses diffuse color