updated test_class for code samples