update vertex shader and add per vertex color