refactor keyboard focus in lazy way