Add ColorGradient class