Update base_unit.rb