diff --git a/academic_codes/2020.11.04_Chern_number_of_the_cross_section_plane_in_Weyl_semimetals/Chern_number_of_the_cross_section_plane_in_Weyl_semimetals_with_guan_package.py b/academic_codes/2020.11.04_Chern_number_of_the_cross_section_plane_in_Weyl_semimetals/Chern_number_of_the_cross_section_plane_in_Weyl_semimetals_with_guan_package.py
index 9556d5e..f4c1369 100644
--- a/academic_codes/2020.11.04_Chern_number_of_the_cross_section_plane_in_Weyl_semimetals/Chern_number_of_the_cross_section_plane_in_Weyl_semimetals_with_guan_package.py
+++ b/academic_codes/2020.11.04_Chern_number_of_the_cross_section_plane_in_Weyl_semimetals/Chern_number_of_the_cross_section_plane_in_Weyl_semimetals_with_guan_package.py
@@ -13,7 +13,7 @@ def main():
     for kz in kz_array:
         print(kz)
         hamiltonian_function = functools.partial(hamiltonian, kz=kz)
-        chern_number = guan.calculate_chern_number_for_square_lattice(hamiltonian_function)
+        chern_number = guan.calculate_chern_number_for_square_lattice_with_efficient_method(hamiltonian_function)
         chern_number_array.append(chern_number)
     guan.plot(kz_array, chern_number_array, style='-o')
 
@@ -27,4 +27,4 @@ def hamiltonian(kx,ky,kz):  # Weyl semimetal
 
 
 if __name__ == '__main__':
-    main()
+    main()
\ No newline at end of file
diff --git a/academic_codes/2021.12.27_Chern_numbers_of_Landau_levels/Chern_numbers_of_Landau_levels_in_square_lattice.py b/academic_codes/2021.12.27_Chern_numbers_of_Landau_levels/Chern_numbers_of_Landau_levels_in_square_lattice.py
index 1d08bc9..cb51608 100644
--- a/academic_codes/2021.12.27_Chern_numbers_of_Landau_levels/Chern_numbers_of_Landau_levels_in_square_lattice.py
+++ b/academic_codes/2021.12.27_Chern_numbers_of_Landau_levels/Chern_numbers_of_Landau_levels_in_square_lattice.py
@@ -33,7 +33,7 @@ def main():
     guan.plot(k_array, eigenvalue_array, xlabel='kx', ylabel='E', style='k')
 
     H_k = functools.partial(hamiltonian, Ny=Ny, B=1/Ny)
-    chern_number = guan.calculate_chern_number_for_square_lattice(H_k, precision=100)
+    chern_number = guan.calculate_chern_number_for_square_lattice_with_efficient_method(H_k, precision=100)
     print(chern_number)
     print(sum(chern_number))