Add tests for update method