Added end screen and ending condition