Add Connect Four code