Add tests for the update method