Add tests for class prefixing