Add button methods to update