Handling tab focus correctly