Cleaned up Connect Four code