Modified layout and added end game prompt