add handling for game over case plus the way to restart app