Added more test cases for update method