diff --git a/UnMicst1-5.py b/UnMicst1-5.py
index 193e98b..5d180d0 100644
--- a/UnMicst1-5.py
+++ b/UnMicst1-5.py
@@ -820,6 +820,7 @@ def singleImageInference(image, mode, pmIndex):
 		'metadata': None,
 		'append': False,
 	}
+	print(nClass)
 	if args.stackOutput:
 		slice = 0
 		for iClass in args.classOrder[::-1]:
@@ -841,6 +842,7 @@ def singleImageInference(image, mode, pmIndex):
 				skimage.io.imsave(args.outputPath +  '//qc//' + fileNamePrefix[0] + '_Preview_' + str(dapiChannel) + '.tif',
 								  np.uint8(255 * rawI), **append_kwargs)
 			slice = slice + 1
+			print(slice)
 
 	else:
 		contours = np.uint8(255 * UNet2D.singleImageInference(cells, 'accumulate', args.classOrder[1]))