Update test_class.lua