Add testing code to file