Task#6 is not clear, I just fixed the code to use color parameter properly