Added util function to store test classes