Update input focus handling