diff --git a/src/cdpy/dw.py b/src/cdpy/dw.py
index d4ef5e5..6bcc692 100644
--- a/src/cdpy/dw.py
+++ b/src/cdpy/dw.py
@@ -143,7 +143,8 @@ def create_cluster(self, env_crn: str, overlay: bool, aws_lb_subnets: list = Non
             ]
         )
 
-    def create_data_visualization(self, cluster_id: str, name: str, config: dict = None):
+    def create_data_visualization(self, cluster_id: str, name: str, config: dict = None,
+                                   template_name: str = None, image_version: str = None):
         return self.sdk.call(
             svc='dw', func='create_data_visualization', ret_field='dataVisualizationId',
             squelch=[
@@ -152,6 +153,8 @@ def create_data_visualization(self, cluster_id: str, name: str, config: dict = N
             clusterId=cluster_id,
             name=name,
             config=config,
+            templateName = template_name,
+            imageVersion = image_version
         )
 
     def delete_cluster(self, cluster_id: str, force: bool = False):