Add testing code.