Set focus on button/input.
Return focus to activeElement on button click.