Update main.py

set focus at widget tree after add widget.