Rewrite the scheduler in an agent-centric way

This seems to perform better than the old one.
Could we name this something other than `i`?