Correct colour codes a tad