Slight changes to connect four case.