diff --git a/README.md b/README.md
index 72df406..f45ab4d 100644
--- a/README.md
+++ b/README.md
@@ -21,3 +21,17 @@ Can also be used with the [bem function](https://github.com/drupal-pattern-lab/b
 
 <div {{ add_attributes(additional_attributes) }}></div>
 ```
+
+Can also be used with title_attributes or content_attributes:
+```
+{% set additional_title_attributes = {
+  "class": ["foo__title", "bar__title"],
+} %}
+{% set additional_content_attributes = {
+  "class": ["foo__content", "bar__content"],
+} %}
+
+<h2 {{ add_attributes(additional_title_attributes,'title_attributes') }}></h2>
+<div {{ add_attributes(additional_content_attributes,'content_attributes') }}>
+
+```
\ No newline at end of file
diff --git a/add_attributes.function.php b/add_attributes.function.php
index 9e95113..2d41632 100644
--- a/add_attributes.function.php
+++ b/add_attributes.function.php
@@ -6,7 +6,11 @@
 
 use Drupal\Core\Template\Attribute;
 
-$function = new Twig_SimpleFunction('add_attributes', function ($context, $additional_attributes = []) {
+$function = new Twig_SimpleFunction('add_attributes', function ($context, $additional_attributes = [], $attribute_type = 'attributes') {
+  if (!in_array($attribute_type, ['attributes','title_attributes','content_attributes'])) {
+    throw new Exception('Invalid attribute type.');
+  }
+
   if (class_exists('Drupal')) {
     $attributes = new Attribute();
 
@@ -35,21 +39,21 @@
           }
         }
         // Merge additional attribute values with existing ones.
-        if ($context['attributes']->offsetExists($key)) {
-          $existing_attribute = $context['attributes']->offsetGet($key)->value();
+        if ($context[$attribute_type]->offsetExists($key)) {
+          $existing_attribute = $context[$attribute_type]->offsetGet($key)->value();
           $value = array_merge($existing_attribute, $value);
         }
 
-        $context['attributes']->setAttribute($key, $value);
+        $context[$attribute_type]->setAttribute($key, $value);
       }
     }
 
     // Set all attributes.
-    foreach($context['attributes'] as $key => $value) {
+    foreach($context[$attribute_type] as $key => $value) {
       $attributes->setAttribute($key, $value);
       // Remove this attribute from context so it doesn't filter down to child
       // elements.
-      $context['attributes']->removeAttribute($key);
+      $context[$attribute_type]->removeAttribute($key);
     }
 
     return $attributes;