OK, I take it back - it was quite easy to incorporate it in optimize... but I had to change evaluate etc. in the optimization tests. Anyway, now it reflects Map/Reduce more in a way which is great.