moved color handling to the vertex shader