.

1 files changed, 3 insertions, 2 deletions

diff --git a/lib/camera/camera.rb b/lib/camera/camera.rb
index e5fd254..cfea114 100644
--- a/lib/camera/camera.rb
+++ b/lib/camera/camera.rb
@@ -36,9 +36,10 @@ module Camera
     case item
     when Triangle
       item.extend Camera::TriangleWrapped
-      puts item.class
     when Square
       item.extend Camera::SquareWrapped
+    when Rectangle
+      item.extend Camera::RectangleWrapped
     when Quad
       item.extend Camera::QuadWrapped
     when Line
@@ -53,7 +54,7 @@ module Camera
       item.extend Camera::TextWrapped
     else
       puts 'Warning: Non-standard Object added to Camera'
-      puts 'Object may not behave as expected'
+      puts '  Object may not behave as expected'
     end
     objects.push(item) unless objects.include?(item)
   end