Changed callback to only happen on buttons.