#11 Added end game