diff --git a/diffbot_base/src/diffbot_base.cpp b/diffbot_base/src/diffbot_base.cpp
index 1e9ea626..c3413d6a 100644
--- a/diffbot_base/src/diffbot_base.cpp
+++ b/diffbot_base/src/diffbot_base.cpp
@@ -1,5 +1,6 @@
 #include <ros/ros.h>
 #include <diffbot_base/diffbot_hw_interface.h>
+#include<iostrem.h>
 #include <controller_manager/controller_manager.h>
  
 int main(int argc, char **argv)
diff --git a/diffbot_description/urdf/include/addon_macro.urdf.xacro b/diffbot_description/urdf/include/addon_macro.urdf.xacro
index 10437b7b..73340984 100644
--- a/diffbot_description/urdf/include/addon_macro.urdf.xacro
+++ b/diffbot_description/urdf/include/addon_macro.urdf.xacro
@@ -25,7 +25,7 @@
           <!-- The folder path get a folder_name global parameter -->
           <mesh filename="package://${package_name}/meshes/${robot_name}/${base_prop[addon]['mesh']}" scale="${base_prop[addon]['scale']}"/>
         </geometry>
-        <material name="white" />
+        <material name="red" />
       </visual>
     </xacro:if>
     <xacro:if value="${base_prop[addon]['mesh'] == '' }" >