diff --git a/lib/fastlane/plugin/test_report/actions/test_report_action.rb b/lib/fastlane/plugin/test_report/actions/test_report_action.rb
index 2bc5eb8..92304cb 100644
--- a/lib/fastlane/plugin/test_report/actions/test_report_action.rb
+++ b/lib/fastlane/plugin/test_report/actions/test_report_action.rb
@@ -1,7 +1,7 @@
 require 'fastlane/action'
 require_relative '../helper/test_report_helper'
 require "erb"
-require "nokogiri"
+require "rexml/document"
 
 module Fastlane
   module Actions
@@ -9,45 +9,57 @@ class TestReportAction < Action
       def self.run(params)
         UI.message("The test_report plugin is working!")
 
-        report = File.read(File.expand_path(params[:path]))
-        doc = Nokogiri::XML(File.open(report))
-        test_suite_name = @doc.xpath('//test_suite').attr("name")
+        include REXML
+
+        File.rename("fastlane/test_output/report.junit", "fastlane/test_output/report.xml")
+
+        file = File.new(File.expand_path(params[:report_path]))
+        doc = Document.new(file)
 
         template = '---
-        layout: testReport
-        ---
+layout: testReport
+---
+                
+                <div class="total row">
+                <h2 class="col-sm-18">Test Results</h2>
+                <h2 class="col-sm-6 text-right"><%= doc.root.attributes["tests"] %> tests</h2>
+            </div>
+            <div>
+            <% i = 0 %>
+                <% doc.elements.each("testsuites/testsuite") do |name| %>
+                    <% i = i + 1 %>
+                    <% if name.attributes["failures"] == "0" %>
+                <div class="test-suite row test-suite--passing" onclick="changeDisplay(<%= "\'#test-#{i}\'"%>)">
+                    <% else %>
+                <div class="test-suite row test-suite--failing" onclick="changeDisplay(<%= "\'#test-#{i}\'"%>)">
+                    <% end %>
+                    <h4 class="col-sm-20"><%= name.attributes["name"] %></h4>
+                    <h4 class="col-sm-4 text-right"><%= name.attributes["tests"] %> tests</h4>
+                </div>
+                <!-- Failing or passing class -->
+                <div id="<%= "test-#{i}"%>">
+                 
+                    <% doc.elements.each("testsuites/testsuite/testcase") do |test| %>
+                        <% if test.attributes["classname"] == name.attributes["name"] %>
+                        <div class="test-case--passing row">
+                         <div class="col-sm-20">
+                             <p><%= test.attributes["name"] %></p>
+                         </div>
+                         <div class="col-sm-4 text-right">
+                             <p><%= test.attributes["time"] %></p>
+                         </div>
+                        </div>
+                        <% end %>
+                    <% end %>
+                </div>
+                 <% end %>
         
-        <div class="total row">
-    <div class="col-sm-12">
-        <h2>Test Results</h2>
-    </div>
-    <div class="col-sm-12">
-        <h2> 8 tests</h2>
-    </div>
-</div>
-<% test_suite_name.each do |name| %>
-<div>
-    <div class="test-suite row failing" onclick="changeDisplay('test-1')">
-        <h4 class="col-sm-12"><%= name %></h4>
-        <h4 class="col-sm-12">1 tests</h4>
-    </div>
-<% end %>
-    <div id="test-1">
+            </div>'
 
-        <div class="failing row">
-            <div class="title col-sm-12 dropdown-item">
-                <p>testTakeScreenshots</p>
-            </div>
-            <div class="time col-sm-12 dropdown-item">
-                <p>29.457s</p>
-            </div>
-        </div>
-    </div>
-</div>'
 
         result = ERB.new(template).result(binding())
 
-        open('test-report/index.html', 'w') do |f|
+        open('fastlane/test_output/index.html', 'w') do |f|
           f.puts result
         end