Fixed a problem with EndScreen not showing.
Merge pull request #10 from dreisch/restart_endscreen

Fixed a problem with EndScreen not showing.