Added update button with function