Updated input box implementation: editing follows focus