Training with diverse class

Training with diverse class