Update test_class.rb