Improve the algorithm to make it work with more agents
Improve the algorithm to make it work with more agents