Add return focus to input area on action