Add autofocus to input element