update test and add id function