Updated the test with an example for two agents.