diff --git a/.gitignore b/.gitignore
index 3beccbc..68076b9 100644
--- a/.gitignore
+++ b/.gitignore
@@ -16,4 +16,6 @@ test/tmp
 test/version_tmp
 tmp
 *.swp
-demo/graph.html
+demo/graph*.html
+demo/data.json
+tags
diff --git a/Gemfile b/Gemfile
index a000450..07973fd 100644
--- a/Gemfile
+++ b/Gemfile
@@ -5,5 +5,5 @@ gemspec
 
 # both are optional, depending on platform
 gem 'fast_stack'
-gem 'stackprof', platform: [:mri_21, :mri_22, :mri_23]
+gem 'stackprof', platform: [:mri_21, :mri_22, :mri_23, :mri_24]
 
diff --git a/README.md b/README.md
index c5dab55..fc6495f 100644
--- a/README.md
+++ b/README.md
@@ -58,6 +58,8 @@ http://samsaffron.github.io/flamegraph/rails-startup.html
 
 1. Fork it
 2. Create your feature branch (`git checkout -b my-new-feature`)
-3. Commit your changes (`git commit -am 'Add some feature'`)
-4. Push to the branch (`git push origin my-new-feature`)
-5. Create new Pull Request
+3. Make your changes
+4. Run the tests (`bundle exec ruby -Ilib:test -e 'ARGV.each { |f| require f }' ./test/test*.rb`)
+5. Commit your changes (`git commit -am 'Add some feature'`)
+6. Push to the branch (`git push origin my-new-feature`)
+7. Create new Pull Request
diff --git a/lib/flamegraph/flamegraph.html b/lib/flamegraph/flamegraph.html.erb
similarity index 98%
rename from lib/flamegraph/flamegraph.html
rename to lib/flamegraph/flamegraph.html.erb
index 175fc25..5555412 100644
--- a/lib/flamegraph/flamegraph.html
+++ b/lib/flamegraph/flamegraph.html.erb
@@ -74,7 +74,18 @@ <h3>Frame Info</h3>
   </div>
   <script>
 
-var data = /**DATA**/;
+var data = null;
+<% if ENV['TEST'] %>
+  $.ajax({
+    url: './<%= ENV['FILE'] || 'data.json' %>',
+    async: false,
+    success: function(response) {
+      data = JSON.parse(response);
+    }
+  });
+<% else %>
+  data = /**DATA**/;
+<% end %>
 var maxX = 0;
 var maxY = 0;
 
diff --git a/lib/flamegraph/renderer.rb b/lib/flamegraph/renderer.rb
index 5ca4c69..60249b7 100644
--- a/lib/flamegraph/renderer.rb
+++ b/lib/flamegraph/renderer.rb
@@ -1,5 +1,6 @@
 # inspired by https://github.com/brendangregg/FlameGraph
 require 'base64'
+require 'erb'
 
 class Flamegraph::Renderer
   def initialize(stacks)
@@ -7,7 +8,7 @@ def initialize(stacks)
   end
 
   def graph_html(embed_resources)
-    body = read('flamegraph.html')
+    body = ERB.new(read('flamegraph.html.erb')).result
     body.sub! "/**INCLUDES**/",
       if embed_resources
         embed("jquery.min.js","d3.min.js","lodash.min.js")
diff --git a/test/test_helper.rb b/test/test_helper.rb
index e7a3bfe..547bf5a 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -1,4 +1,3 @@
 require 'flamegraph'
+require 'minitest/autorun'
 require 'minitest/pride'
-
-