Improved input focus