how to program connect 4 in python