Trainable class is working