Adding the testing framework