Added the framework for testing