Updated to work with rainbow gem v2.0.0