end screen works and allows you to restart