Moveing the button update code to the button object