Add autofocus to input box