Updated the class.rb file