Add tests for extensions