diff --git a/README.md b/README.md
index 07f3a28..15a9fca 100644
--- a/README.md
+++ b/README.md
@@ -48,10 +48,10 @@ if __name__ == "__main__":
     proto_file = MessageProto.generate_proto("messageservice", "message_service")
     # generate_pb2(`proto file`)
     MessageProto.generate_pb2(proto_file)
-    # generate_wsagger(`proto file`, `version name`)
-    swagger_file = MessageProto.generate_swagger(proto_file, "2.0.0")
-    # run swagger with Flask
-    MessageProto.run_swagger()
+    # generate_wsagger(`proto file`)
+    swagger_file = MessageProto.generate_swagger(proto_file)
+    # run_swagger(`version`, `port`)
+    MessageProto.run_swagger(version="2.0.1", port=5937)
 ```
 
 This script will generate a .proto file in the 'protos/' directory and pb2 files in the 'outputs/' directory.