clean up tests a little
Test update functionality