Update connect_four.py