need to collect labels for return

so the interpreter can find the object

lib/register/collector.rb

@@ -13,6 +13,10 @@ module Register
       #puts "adding #{' ' * depth}:#{object.class}"
       #puts "ADD #{object.first.class}, #{object.last.class}" if object.is_a? Array
       return unless self.add_object object
+      # probably should make labels or even instructions derive from Parfait::Object, but . .
+      if  object.is_a? Register::Label
+        object.each_label { |l| self.add_object(l)}
+      end
       return unless object.respond_to? :has_layout?
       layout = object.get_layout
       keep(layout  , depth + 1)

lib/register/machine.rb

@@ -55,8 +55,9 @@ module Register
     def add_object o
       return false if @objects[o.object_id]
       return true if o.is_a? Fixnum
-      return true if o.is_a? Register::Label
-      raise "adding non parfait #{o.class}" unless o.is_a? Parfait::Object or o.is_a? Symbol
+      unless o.is_a? Parfait::Object or o.is_a? Symbol or o.is_a? Register::Label
+        raise "adding non parfait #{o.class}"
+      end
       @objects[o.object_id] = o
       true
     end