clean up on full working single label classification