Ant colony optimization algorithm in Python