a few changes to end game