Add testing code, need debug