fixed some bugs with the classification generator