Add base helper class for test cases