Small fix of colours adding and removing from a class