Update class.rb

Please accept my changes!

It makes it all work