Added "test function" class