added color correction