Thanks. A few diagrams on the spatial layout and information flow between stages would indeed go a long way. I'll take that up in the upcoming commits.