Add testing functionality