diff --git a/global/overlay/etc/puppet/modules/eid/manifests/telegraf2.pp b/global/overlay/etc/puppet/modules/eid/manifests/telegraf2.pp
index 1c3099bd..6d1a4886 100644
--- a/global/overlay/etc/puppet/modules/eid/manifests/telegraf2.pp
+++ b/global/overlay/etc/puppet/modules/eid/manifests/telegraf2.pp
@@ -1,10 +1,14 @@
 # This puppet manifest is used to configure Telegraf for Sweden Connect
 
-# @param version              Version of the docker image to use. (referenced in compose file)
-# @param server_fqdn          The FQDN of the server. (referenced in compose file)
-# @param ports                Port mappings for container
-# @param service_dir          The directory where all related config and files are stored. (referenced in compose file)
-# @param influx_urls          List of influxdb urls to send data to
+# @param version                 Version of the docker image to use. (referenced in compose file)
+# @param server_fqdn             The FQDN of the server. (referenced in compose file)
+# @param server_type             Type of server, used in compose for config differentiation
+# @param ports                   Port mappings for container
+# @param service_dir             The directory where all related config and files are stored. (referenced in compose file)
+# @param influx_urls             List of influxdb urls to send data to
+# @param influxdb2_org           Influx organization (referenced in compose file)
+# @param influxdb2_bucket        Influx bucket name (referenced in compose file)
+# @param influxdb2_bucket_proxy  Influx bucket name (referenced in compose file)
 class eid::telegraf2 (
   String $version                    = 'latest',
   String $server_fqdn                = $facts['networking']['fqdn'],
@@ -29,6 +33,7 @@ class eid::telegraf2 (
       description      => 'telegraf'
     }
 
+    # Create the influx config file
     file { "${service_dir}/telegraf.conf":
       ensure  => file,
       owner   => 'root',