diff --git a/slakonet/main.py b/slakonet/main.py
index 58c0d17..1b42f7a 100644
--- a/slakonet/main.py
+++ b/slakonet/main.py
@@ -3244,4 +3244,4 @@ def print_energy(a=atoms):
     fig_bands, ax_bands = calc_bands.plot_band_structure(
         fermi_shift=True, save_path="bands_enhanced.png"
     )
-    plt.show()
\ No newline at end of file
+    plt.show()
diff --git a/slakonet/optim.py b/slakonet/optim.py
index 07a4aff..34d18a9 100644
--- a/slakonet/optim.py
+++ b/slakonet/optim.py
@@ -3325,4 +3325,4 @@ def get_hamiltonian(jarvis_atoms=None, kpts=[1, 4, 4], model=None):
     # )
     multi_vasp_training()
 
-# """
\ No newline at end of file
+# """
diff --git a/slakonet/train_slakonet.py b/slakonet/train_slakonet.py
index 22ce98b..8b0d540 100644
--- a/slakonet/train_slakonet.py
+++ b/slakonet/train_slakonet.py
@@ -106,4 +106,4 @@ class SlakoNetConfig(BaseSettings):
         bandgap_weight=config.bandgap_weight,
     )
 
-    print("\n✅ Training completed successfully!")
\ No newline at end of file
+    print("\n✅ Training completed successfully!")