Added update method