Added end game & clean up stuff