now uses class labels