Set/remove focus of an input with a button