Add testing for udpate