Update register_class.rb