diff --git a/modules/deeplearning/cloud_fraction_fcn_abi.py b/modules/deeplearning/cloud_fraction_fcn_abi.py
index d8fc9e3a230ac08aefb4dea03f7c2405c323b51d..56727b5c8e493f9109a6b810f6ad0270215da462 100644
--- a/modules/deeplearning/cloud_fraction_fcn_abi.py
+++ b/modules/deeplearning/cloud_fraction_fcn_abi.py
@@ -20,8 +20,8 @@ LOG_DEVICE_PLACEMENT = False
 PROC_BATCH_SIZE = 4
 PROC_BATCH_BUFFER_SIZE = 5000
 
-NumClasses = 7
-# NumClasses = 5
+# NumClasses = 7
+NumClasses = 5
 if NumClasses == 2:
     NumLogits = 1
 else:
@@ -355,16 +355,6 @@ class SRCNN:
             tmp = normalize(tmp, param, mean_std_dct)
             data_norm.append(tmp)
 
-        # for param in sub_fields:
-        #     idx = params.index(param)
-        #     tmp = input_data[:, idx, :, :]
-        #     tmp = tmp[:, slc_y, slc_x]
-        #     if param != 'refl_substddev_ch01':
-        #         tmp = normalize(tmp, 'refl_0_65um_nom', mean_std_dct)
-        #     else:
-        #         tmp = np.where(np.isnan(tmp), 0, tmp)
-        #     data_norm.append(tmp)
-
         rlo = input_data[:, params.index('refl_submin_ch01'), :, :]
         rlo = rlo[:, slc_y, slc_x]
         rlo = normalize(rlo, 'refl_0_65um_nom', mean_std_dct)