Improve input auto-focusing