Remove additional code for button updating