chore(buttons): cleanup code