Add testing function