Added comments and changed color vec3 to vec4