Adding ant colony AI code