diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index 0b8a0b0..09edd87 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -2,7 +2,9 @@
 <project version="4">
   <component name="ChangeListManager">
     <list default="true" id="1e2e223d-dea0-427d-a42c-c5910de29136" name="Default Changelist" comment="">
+      <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/InferenceProcessor.java" beforeDir="false" afterPath="$PROJECT_DIR$/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/InferenceProcessor.java" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/SSDClassifierService.java" beforeDir="false" afterPath="$PROJECT_DIR$/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/SSDClassifierService.java" afterDir="false" />
     </list>
     <ignored path="$PROJECT_DIR$/out/" />
     <ignored path="$PROJECT_DIR$/target/" />
@@ -21,11 +23,23 @@
   </component>
   <component name="FileEditorManager">
     <leaf>
-      <file pinned="false" current-in-tab="true">
+      <file pinned="false" current-in-tab="false">
         <entry file="file://$PROJECT_DIR$/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/InferenceProcessor.java">
           <provider selected="true" editor-type-id="text-editor">
-            <state relative-caret-position="252">
-              <caret line="169" lean-forward="true" selection-start-line="169" selection-end-line="169" />
+            <state relative-caret-position="329">
+              <caret line="162" column="54" selection-start-line="162" selection-start-column="54" selection-end-line="162" selection-end-column="54" />
+              <folding>
+                <element signature="imports" expanded="true" />
+              </folding>
+            </state>
+          </provider>
+        </entry>
+      </file>
+      <file pinned="false" current-in-tab="true">
+        <entry file="file://$PROJECT_DIR$/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/SSDClassifierService.java">
+          <provider selected="true" editor-type-id="text-editor">
+            <state relative-caret-position="262">
+              <caret line="139" column="28" selection-start-line="139" selection-start-column="28" selection-end-line="139" selection-end-column="28" />
               <folding>
                 <element signature="imports" expanded="true" />
               </folding>
@@ -69,9 +83,9 @@
         <option value="$PROJECT_DIR$/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/SSDClassifier.java" />
         <option value="$PROJECT_DIR$/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/Result.java" />
         <option value="$PROJECT_DIR$/nifi-mxnetinference-processors/pom.xml" />
-        <option value="$PROJECT_DIR$/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/SSDClassifierService.java" />
         <option value="$PROJECT_DIR$/nifi-mxnetinference-processors/src/test/java/com/dataflowdeveloper/mxnet/InferenceProcessorTest.java" />
         <option value="$PROJECT_DIR$/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/InferenceProcessor.java" />
+        <option value="$PROJECT_DIR$/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/SSDClassifierService.java" />
       </list>
     </option>
   </component>
@@ -168,7 +182,7 @@
     </panes>
   </component>
   <component name="PropertiesComponent">
-    <property name="com.android.tools.idea.instantapp.provision.ProvisionBeforeRunTaskProvider.myTimeStamp" value="1544035056632" />
+    <property name="com.android.tools.idea.instantapp.provision.ProvisionBeforeRunTaskProvider.myTimeStamp" value="1544039782590" />
     <property name="com.intellij.testIntegration.createTest.CreateTestDialog.defaultLibrary" value="JUnit4" />
     <property name="com.intellij.testIntegration.createTest.CreateTestDialog.defaultLibrarySuperClass.JUnit4" value="" />
     <property name="settings.editor.selected.configurable" value="project.propVCSSupport.Mappings" />
@@ -297,34 +311,34 @@
     <servers />
   </component>
   <component name="TestHistory">
-    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 09h 15m 42s.xml">
+    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 10h 43m 25s.xml">
       <configuration name="InferenceProcessorTest.testProcessor" configurationId="JUnit" />
     </history-entry>
-    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 09h 17m 12s.xml">
+    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 13h 26m 26s.xml">
       <configuration name="InferenceProcessorTest.testProcessor" configurationId="JUnit" />
     </history-entry>
-    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 09h 41m 10s.xml">
+    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 13h 37m 46s.xml">
       <configuration name="InferenceProcessorTest.testProcessor" configurationId="JUnit" />
     </history-entry>
-    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 09h 56m 38s.xml">
+    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 14h 28m 40s.xml">
       <configuration name="InferenceProcessorTest.testProcessor" configurationId="JUnit" />
     </history-entry>
-    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 09h 57m 44s.xml">
+    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 14h 33m 48s.xml">
       <configuration name="InferenceProcessorTest.testProcessor" configurationId="JUnit" />
     </history-entry>
-    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 10h 40m 47s.xml">
+    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 14h 36m 14s.xml">
       <configuration name="InferenceProcessorTest.testProcessor" configurationId="JUnit" />
     </history-entry>
-    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 10h 41m 06s.xml">
+    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 14h 36m 56s.xml">
       <configuration name="InferenceProcessorTest.testProcessor" configurationId="JUnit" />
     </history-entry>
-    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 10h 42m 04s.xml">
+    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 14h 38m 00s.xml">
       <configuration name="InferenceProcessorTest.testProcessor" configurationId="JUnit" />
     </history-entry>
-    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 10h 43m 25s.xml">
+    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 14h 43m 27s.xml">
       <configuration name="InferenceProcessorTest.testProcessor" configurationId="JUnit" />
     </history-entry>
-    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 13h 26m 26s.xml">
+    <history-entry file="InferenceProcessorTest_testProcessor - 2018.12.05 at 14h 51m 35s.xml">
       <configuration name="InferenceProcessorTest.testProcessor" configurationId="JUnit" />
     </history-entry>
   </component>
@@ -332,7 +346,7 @@
     <frame x="0" y="22" width="1440" height="810" extended-state="6" />
     <editor active="true" />
     <layout>
-      <window_info active="true" content_ui="combo" id="Project" order="0" visible="true" weight="0.24749643" />
+      <window_info content_ui="combo" id="Project" order="0" visible="true" weight="0.24749643" />
       <window_info id="Structure" order="1" side_tool="true" weight="0.25" />
       <window_info id="Image Layers" order="2" />
       <window_info id="Designer" order="3" />
@@ -341,7 +355,7 @@
       <window_info id="Favorites" order="6" side_tool="true" />
       <window_info anchor="bottom" id="Message" order="0" />
       <window_info anchor="bottom" id="Find" order="1" weight="0.3286908" />
-      <window_info anchor="bottom" id="Run" order="2" visible="true" weight="0.32729805" />
+      <window_info active="true" anchor="bottom" id="Run" order="2" visible="true" weight="0.32729805" />
       <window_info anchor="bottom" id="Debug" order="3" weight="0.4" />
       <window_info anchor="bottom" id="Cvs" order="4" weight="0.25" />
       <window_info anchor="bottom" id="Inspection" order="5" weight="0.4" />
@@ -369,54 +383,6 @@
   <component name="editorHistoryManager">
     <entry file="file://$PROJECT_DIR$/src/main/java/com/dataflowdeveloper/mxnet/App.java" />
     <entry file="file://$PROJECT_DIR$/src/test/java/com/dataflowdeveloper/mxnet/AppTest.java" />
-    <entry file="file://$PROJECT_DIR$/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/Result.java">
-      <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="30">
-          <caret line="83" column="18" selection-start-line="83" selection-start-column="18" selection-end-line="83" selection-end-column="18" />
-          <folding>
-            <element signature="imports" expanded="true" />
-            <element signature="e#645#646#0" expanded="true" />
-            <element signature="e#673#674#0" expanded="true" />
-            <element signature="e#712#713#0" expanded="true" />
-            <element signature="e#746#747#0" expanded="true" />
-            <element signature="e#776#777#0" expanded="true" />
-            <element signature="e#805#806#0" expanded="true" />
-            <element signature="e#846#847#0" expanded="true" />
-            <element signature="e#882#883#0" expanded="true" />
-            <element signature="e#566#567#0" expanded="true" />
-            <element signature="e#588#589#0" expanded="true" />
-            <element signature="e#1118#1119#0" expanded="true" />
-            <element signature="e#1146#1147#0" expanded="true" />
-            <element signature="e#1188#1189#0" expanded="true" />
-            <element signature="e#1222#1223#0" expanded="true" />
-            <element signature="e#1259#1260#0" expanded="true" />
-            <element signature="e#1293#1294#0" expanded="true" />
-            <element signature="e#1346#1347#0" expanded="true" />
-            <element signature="e#1392#1393#0" expanded="true" />
-            <element signature="e#1420#1421#0" expanded="true" />
-            <element signature="e#1447#1448#0" expanded="true" />
-            <element signature="e#1484#1485#0" expanded="true" />
-            <element signature="e#1516#1517#0" expanded="true" />
-            <element signature="e#1546#1547#0" expanded="true" />
-            <element signature="e#1573#1574#0" expanded="true" />
-            <element signature="e#1612#1613#0" expanded="true" />
-            <element signature="e#1644#1645#0" expanded="true" />
-            <element signature="e#1674#1675#0" expanded="true" />
-            <element signature="e#1701#1702#0" expanded="true" />
-            <element signature="e#1740#1741#0" expanded="true" />
-            <element signature="e#1772#1773#0" expanded="true" />
-            <element signature="e#1802#1803#0" expanded="true" />
-            <element signature="e#1829#1830#0" expanded="true" />
-            <element signature="e#1868#1869#0" expanded="true" />
-            <element signature="e#1900#1901#0" expanded="true" />
-            <element signature="e#1930#1931#0" expanded="true" />
-            <element signature="e#1957#1958#0" expanded="true" />
-            <element signature="e#1996#1997#0" expanded="true" />
-            <element signature="e#2028#2029#0" expanded="true" />
-          </folding>
-        </state>
-      </provider>
-    </entry>
     <entry file="file://$PROJECT_DIR$/pom.xml">
       <provider selected="true" editor-type-id="text-editor">
         <state relative-caret-position="201">
@@ -431,12 +397,20 @@
         </state>
       </provider>
     </entry>
-    <entry file="file://$PROJECT_DIR$/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/SSDClassifierService.java">
+    <entry file="file://$PROJECT_DIR$/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/Result.java">
       <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="402">
-          <caret line="147" column="51" selection-start-line="147" selection-start-column="51" selection-end-line="147" selection-end-column="51" />
+        <state relative-caret-position="-287">
+          <caret line="83" column="18" selection-start-line="83" selection-start-column="18" selection-end-line="83" selection-end-column="18" />
           <folding>
             <element signature="imports" expanded="true" />
+            <element signature="e#645#646#0" expanded="true" />
+            <element signature="e#673#674#0" expanded="true" />
+            <element signature="e#712#713#0" expanded="true" />
+            <element signature="e#746#747#0" expanded="true" />
+            <element signature="e#776#777#0" expanded="true" />
+            <element signature="e#805#806#0" expanded="true" />
+            <element signature="e#846#847#0" expanded="true" />
+            <element signature="e#882#883#0" expanded="true" />
           </folding>
         </state>
       </provider>
@@ -450,8 +424,18 @@
     </entry>
     <entry file="file://$PROJECT_DIR$/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/InferenceProcessor.java">
       <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="252">
-          <caret line="169" lean-forward="true" selection-start-line="169" selection-end-line="169" />
+        <state relative-caret-position="329">
+          <caret line="162" column="54" selection-start-line="162" selection-start-column="54" selection-end-line="162" selection-end-column="54" />
+          <folding>
+            <element signature="imports" expanded="true" />
+          </folding>
+        </state>
+      </provider>
+    </entry>
+    <entry file="file://$PROJECT_DIR$/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/SSDClassifierService.java">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="262">
+          <caret line="139" column="28" selection-start-line="139" selection-start-column="28" selection-end-line="139" selection-end-column="28" />
           <folding>
             <element signature="imports" expanded="true" />
           </folding>
diff --git a/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/InferenceProcessor.java b/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/InferenceProcessor.java
index 2a4c899..6d1e9f6 100644
--- a/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/InferenceProcessor.java
+++ b/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/InferenceProcessor.java
@@ -147,25 +147,22 @@ public void process(final InputStream input, final java.io.OutputStream out) thr
                             for (Result result : results) {
                                 attributes.put(String.format("label_%d", i), result.getLabel() );
                                 attributes.put(String.format("probability_%d",i), String.format("%.2f", result.getProbability()));
-
-                                if ( result.getXmin() > 0) {
-                                    attributes.put(String.format("xmin_%d", i), String.format("%.2f", result.getXmin()));
-                                    attributes.put(String.format("xmax_%d", i), String.format("%.2f", result.getXmax()));
-                                    attributes.put(String.format("ymin_%d", i), String.format("%.2f", result.getYmin()));
-                                    attributes.put(String.format("ymax_%d", i), String.format("%.2f", result.getYmax()));
-                                    attributes.put(String.format("height_%d", i), String.format("%d", result.getHeight()));
-                                    attributes.put(String.format("width_%d", i), String.format("%d", result.getWidth()));
-                                }
+                                attributes.put(String.format("xmin_%d", i), String.format("%.2f", result.getXmin()));
+                                attributes.put(String.format("xmax_%d", i), String.format("%.2f", result.getXmax()));
+                                attributes.put(String.format("ymin_%d", i), String.format("%.2f", result.getYmin()));
+                                attributes.put(String.format("ymax_%d", i), String.format("%.2f", result.getYmax()));
+                                attributes.put(String.format("height_%d", i), String.format("%d", result.getHeight()));
+                                attributes.put(String.format("width_%d", i), String.format("%d", result.getWidth()));
 
                                 Graphics2D g2d = (Graphics2D) img.createGraphics();
 
                                 try {
-                                    // Draw on the buffered image
+                                    // Draw on the buffered image with our rectangles
                                     g2d.setStroke(new BasicStroke(3));
-                                    g2d.setColor(Color.GREEN);
-                                    g2d.drawRect(Math.round(result.getXmin()), Math.round(result.getYmin()),
-                                            Math.round(result.getWidth()),
-                                            Math.round(result.getHeight()));
+                                    g2d.setColor(Color.BLUE);
+                                    g2d.drawRect((int)(result.getXmin()), (int)(result.getYmin()),
+                                            (int)(result.getXmax()),
+                                            (int)(result.getYmax()));
                                     g2d.dispose();
 
                                 } catch (Exception e) {
diff --git a/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/SSDClassifierService.java b/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/SSDClassifierService.java
index 84da2a8..c9e1f13 100644
--- a/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/SSDClassifierService.java
+++ b/nifi-mxnetinference-processors/src/main/java/com/dataflowdeveloper/mxnet/SSDClassifierService.java
@@ -137,13 +137,13 @@ public List<Result> ssdClassify(String modelPath, byte[] imageBytes) {
                     for (ObjectDetectorOutput i : ele) {
                         if ( i != null ) {
                             Result result = new Result();
-
+                            
                             result.setLabel(i.getClassName());
                             result.setProbability(i.getProbability());
-                            result.setXmin(i.getXMin() * width);
-                            result.setXmax(i.getXMax() * height);
-                            result.setYmin(i.getYMin() * width);
-                            result.setYmax(i.getYMax() * height);
+                            result.setXmin(i.getXMin() * width ); //
+                            result.setXmax(i.getXMax() * width ); //
+                            result.setYmin(i.getYMin() * height ); //
+                            result.setYmax(i.getYMax() * height ); //
                             result.setWidth(width);
                             result.setHeight(height);
                             result.setRank(rank);
diff --git a/nifi-mxnetinference-processors/target/classes/com/dataflowdeveloper/mxnet/InferenceProcessor$1.class b/nifi-mxnetinference-processors/target/classes/com/dataflowdeveloper/mxnet/InferenceProcessor$1.class
index e62e858..f0b5bc6 100644
Binary files a/nifi-mxnetinference-processors/target/classes/com/dataflowdeveloper/mxnet/InferenceProcessor$1.class and b/nifi-mxnetinference-processors/target/classes/com/dataflowdeveloper/mxnet/InferenceProcessor$1.class differ
diff --git a/nifi-mxnetinference-processors/target/classes/com/dataflowdeveloper/mxnet/InferenceProcessor.class b/nifi-mxnetinference-processors/target/classes/com/dataflowdeveloper/mxnet/InferenceProcessor.class
index 2a86c29..8bd5d65 100644
Binary files a/nifi-mxnetinference-processors/target/classes/com/dataflowdeveloper/mxnet/InferenceProcessor.class and b/nifi-mxnetinference-processors/target/classes/com/dataflowdeveloper/mxnet/InferenceProcessor.class differ