diff --git a/UFS_Net_train.py b/UFS_Net_train.py
index 35e9b82..cc3b0ca 100644
--- a/UFS_Net_train.py
+++ b/UFS_Net_train.py
@@ -17,6 +17,7 @@
 import copy
 from UFS_Net_architecture import UFSNet
 
+
 batch_size = 64
 
 # Define 8 class labels