+##############################################################################
+# Copyright (c) 2017 Dell EMC and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+
 from builtins import input
 import readline
 readline.parse_and_bind("tab: complete")
 
-content = '''# Copyright (c) 2017 Dell EMC and others.
+content = '''##############################################################################
+# Copyright (c) 2017 Dell EMC and others.
 #
 # All rights reserved. This program and the accompanying materials
 # are made available under the terms of the Apache License, Version 2.0
     storperf-master:
         container_name: "storperf-master"
         image: "opnfv/storperf-master:{storperf_tag}"
-        ports:
-            - "8000:8000"
-        env_file: {ENV_FILE}
-        volumes:
-            - {CARBON_DIR}:/opt/graphite/storage/whisper
+        links:
+            - storperf-graphite
 
     storperf-reporting:
         container_name: "storperf-reporting"
         image: "opnfv/storperf-reporting:{reporting_tag}"
-        ports:
-            - "5080:5000"
 
     storperf-swaggerui:
         container_name: "storperf-swaggerui"
         image: "opnfv/storperf-swaggerui:{swaggerui_tag}"
 
+    storperf-graphite:
+        container_name: "storperf-graphite"
+        image: "opnfv/storperf-graphite:{graphite_tag}"
+        volumes:
+            - {CARBON_DIR}:/opt/graphite/storage/whisper
+
     storperf-httpfrontend:
         container_name: "storperf-httpfrontend"
         image: "opnfv/storperf-httpfrontend:{frontend_tag}"
             - storperf-master
             - storperf-reporting
             - storperf-swaggerui
-
+            - storperf-graphite
 '''
 storeperf_tag = input("Enter image TAG for storperf-master: ") or 'latest'
 assert isinstance(storeperf_tag, str)
 frontend_tag = input("Enter image TAG for frontend: ") or 'latest'
 assert isinstance(frontend_tag, str)
 
+graphite_tag = input("Enter image TAG for graphite: ") or 'latest'
+assert isinstance(graphite_tag, str)
+
 swaggerui_tag = input("Enter image TAG for swaggerui: ") or 'latest'
 assert isinstance(swaggerui_tag, str)
 
 f = open('docker-compose.yaml', 'w')
 f.write(content.format(storperf_tag=storeperf_tag, reporting_tag=reporting_tag,
                        frontend_tag=frontend_tag, swaggerui_tag=swaggerui_tag,
+                       graphite_tag=graphite_tag,
                        CARBON_DIR=carbon_dir, ENV_FILE=env_file))
 
 f.close()
 
     storperf-master:
         container_name: "storperf-master"
         image: "opnfv/storperf-master:${TAG}"
-        ports:
-            - "8000:8000"
-        env_file: ${ENV_FILE}
-        volumes:
-            - ${CARBON_DIR}:/opt/graphite/storage/whisper
+        links:
+            - storperf-graphite
 
     storperf-reporting:
         container_name: "storperf-reporting"
         image: "opnfv/storperf-reporting:${TAG}"
-        ports:
-            - "5080:5000"
 
     storperf-swaggerui:
         container_name: "storperf-swaggerui"
         image: "opnfv/storperf-swaggerui:${TAG}"
 
+    storperf-graphite:
+        container_name: "storperf-graphite"
+        image: "opnfv/storperf-graphite:${TAG}"
+        volumes:
+            - ${CARBON_DIR}:/opt/graphite/storage/whisper
+
     storperf-httpfrontend:
         container_name: "storperf-httpfrontend"
         image: "opnfv/storperf-httpfrontend:${TAG}"
         links:
             - storperf-master
             - storperf-reporting
-            - storperf-swaggerui
\ No newline at end of file
+            - storperf-swaggerui
+            - storperf-graphite