Merge the main code and the testing code