Rewrite the scheduler in an agent-centric way

This seems to perform better than the old one.
nit: newline before this for loop