diff --git a/docs/src/index.md b/docs/src/index.md
index 40d01f2..8c1c2d1 100644
--- a/docs/src/index.md
+++ b/docs/src/index.md
@@ -11,4 +11,11 @@ Documentation for [VUMPSAutoDiff](https://github.com/tangwei94/VUMPSAutoDiff.jl)
 
 ```@autodocs
 Modules = [VUMPSAutoDiff]
+Order = [:type, :function]
+Pages = [
+    "vomps.jl",
+    "toolbox/gauge_fixing.jl",
+    "vumps.jl",
+    #... other files
+]
 ```
diff --git a/src/VUMPSAutoDiff.jl b/src/VUMPSAutoDiff.jl
index 6f33972..7f02df6 100644
--- a/src/VUMPSAutoDiff.jl
+++ b/src/VUMPSAutoDiff.jl
@@ -15,7 +15,9 @@ export MPSMPSTransferMatrix, MPSMPOMPSTransferMatrix
 export ACMap, fixed_point
 export right_env, left_env
 export gauge_fixing, overall_u1_phase
-export mps_update, vumps_update, vumps
+export mps_update!, mps_update, vumps_update, vumps
+
+export vomps!, VOMPSOptions
 
 export DIIS_extrapolation_alg, power_method_alg, iterative_solver
 
@@ -34,4 +36,6 @@ include("vumps.jl");
 include("toolbox/diis.jl")
 include("toolbox/gauge_fixing.jl"); 
 
+include("vomps.jl");
+
 end