Rewrite demo assignment

This is a proposed implementation of the assignment layout discussed in #67 .

Resolves #67 
I'm fine with that. I can add that later, unless you wanted include test-ability in this PR. The main purpose of this PR is just to get a concrete example of the proposal in #67 .