diff --git a/ballerina/Ballerina.toml b/ballerina/Ballerina.toml
index bc9b99a..ba4c6d3 100644
--- a/ballerina/Ballerina.toml
+++ b/ballerina/Ballerina.toml
@@ -3,10 +3,11 @@ authors = ["Ballerina"]
 distribution = "2201.12.0"
 keywords = ["AI", "Agent", "Ollama", "Model", "Provider"]
 license = ["Apache-2.0"]
+icon="icon.png"
 name = "ai.ollama"
 org = "ballerinax"
 repository = "https://github.com/ballerina-platform/module-ballerinax-ai.ollama"
-version = "1.0.1"
+version = "1.1.0"
 
 [platform.java21]
 graalvmCompatible = true
@@ -14,5 +15,5 @@ graalvmCompatible = true
 [[platform.java21.dependency]]
 groupId = "io.ballerina.lib"
 artifactId = "ai.ollama-native"
-version = "1.0.1"
-path = "../native/build/libs/ai.ollama-native-1.0.1-SNAPSHOT.jar"
+version = "1.1.0"
+path = "../native/build/libs/ai.ollama-native-1.1.0-SNAPSHOT.jar"
diff --git a/ballerina/CompilerPlugin.toml b/ballerina/CompilerPlugin.toml
index 40d09fd..2a9bb40 100644
--- a/ballerina/CompilerPlugin.toml
+++ b/ballerina/CompilerPlugin.toml
@@ -3,7 +3,7 @@ id = "ai-ollama-compiler-plugin"
 class = "io.ballerina.lib.ai.ollama.AiOllamaCompilerPlugin"
 
 [[dependency]]
-path = "../compiler-plugin/build/libs/ai.ollama-compiler-plugin-1.0.1-SNAPSHOT.jar"
+path = "../compiler-plugin/build/libs/ai.ollama-compiler-plugin-1.1.0-SNAPSHOT.jar"
 
 [[dependency]]
 path = "../compiler-plugin/build/libs/ballerina-to-openapi-2.3.0.jar"
diff --git a/ballerina/Dependencies.toml b/ballerina/Dependencies.toml
index 8f6f92a..1894cfd 100644
--- a/ballerina/Dependencies.toml
+++ b/ballerina/Dependencies.toml
@@ -5,7 +5,7 @@
 
 [ballerina]
 dependencies-toml-version = "2"
-distribution-version = "2201.12.0"
+distribution-version = "2201.12.7"
 
 [[package]]
 org = "ballerina"
@@ -394,7 +394,7 @@ dependencies = [
 [[package]]
 org = "ballerinax"
 name = "ai.ollama"
-version = "1.0.1"
+version = "1.1.0"
 dependencies = [
 	{org = "ballerina", name = "ai"},
 	{org = "ballerina", name = "constraint"},
diff --git a/ballerina/icon.png b/ballerina/icon.png
new file mode 100644
index 0000000..68d5be8
Binary files /dev/null and b/ballerina/icon.png differ
diff --git a/ballerina/provider.bal b/ballerina/provider.bal
index 7398de7..d239183 100644
--- a/ballerina/provider.bal
+++ b/ballerina/provider.bal
@@ -73,7 +73,8 @@ public isolated client class ModelProvider {
     # + prompt - The prompt to use in the chat messages
     # + td - Type descriptor specifying the expected return type format
     # + return - Generates a value that belongs to the type, or an error if generation fails
-    isolated remote function generate(ai:Prompt prompt, typedesc<anydata> td = <>) returns td|ai:Error = @java:Method {
+    isolated remote function generate(ai:Prompt prompt, @display {label: "Expected type"} typedesc<anydata> td = <>) 
+            returns td|ai:Error = @java:Method {
         'class: "io.ballerina.lib.ai.ollama.Generator"
     } external;
 
diff --git a/build-config/resources/Ballerina.toml b/build-config/resources/Ballerina.toml
index cc391f7..4b48fd6 100644
--- a/build-config/resources/Ballerina.toml
+++ b/build-config/resources/Ballerina.toml
@@ -3,6 +3,7 @@ authors = ["Ballerina"]
 distribution = "2201.12.0"
 keywords = ["AI", "Agent", "Ollama", "Model", "Provider"]
 license = ["Apache-2.0"]
+icon="icon.png"
 name = "ai.ollama"
 org = "ballerinax"
 repository = "https://github.com/ballerina-platform/module-ballerinax-ai.ollama"
diff --git a/gradle.properties b/gradle.properties
index f07536f..5bbc77c 100644
--- a/gradle.properties
+++ b/gradle.properties
@@ -1,6 +1,6 @@
 org.gradle.caching=true
 group=io.ballerina.lib
-version=1.0.1-SNAPSHOT
+version=1.1.0-SNAPSHOT
 ballerinaLangVersion=2201.12.0
 
 shadowJarPluginVersion=8.1.1