Update tests to test for create and update aswell