Add autofocus for input box