fixed bug that didn't allow to return to main menu after game over

added logic to handle disconnects