Enabled different class color based on stereotype