Adding support for extensions also renaming test classes