fix correct_color method - it works!  (i think)