Add listeners to button states