Add ColorGradient(*Model|Label) and clean up code