Update unit.rb