Solved issue with losing focus
"Main menu" button fixed
Restart implemented