Add tests/logic for end of game