Test added for update method