diff --git a/README.md b/README.md
index 23929dd..7299ef9 100644
--- a/README.md
+++ b/README.md
@@ -16,23 +16,19 @@ docker run -v ~/ollama:/root/.ollama -p 11434:11434 ollama/ollama
 Submit a question to Ollama using Ollama4j:
 
 ```java
-public class Test {
-    public static void main(String[] args) throws Exception {
-        String host = "http://localhost:11434/";
-        
-        OllamaAPI ollamaAPI = new OllamaAPI(host);
-        
-        ollamaAPI.pullModel(OllamaModel.LLAMA2);
-        
-        OllamaAsyncResultCallback ollamaAsyncResultCallback = ollamaAPI.runAsync(OllamaModel.LLAMA2, "Who are you?");
-        while (true) {
-            if (ollamaAsyncResultCallback.isComplete()) {
-                System.out.println(ollamaAsyncResultCallback.getResponse());
-                break;
-            }
-            Thread.sleep(1000);
-        }
+String host = "http://localhost:11434/";
+
+OllamaAPI ollamaAPI = new OllamaAPI(host);
+
+ollamaAPI.pullModel(OllamaModel.LLAMA2);
+
+OllamaAsyncResultCallback ollamaAsyncResultCallback = ollamaAPI.runAsync(OllamaModel.LLAMA2, "Who are you?");
+while (true) {
+    if (ollamaAsyncResultCallback.isComplete()) {
+        System.out.println(ollamaAsyncResultCallback.getResponse());
+        break;
     }
+    Thread.sleep(1000);
 }
 ```