Rewrite demo assignment

This is a proposed implementation of the assignment layout discussed in #67 .

Resolves #67 
I don't think this is necessary. If you want to keep it, the demo should do something so that we can test whether it runs.