Rewrite demo assignment

This is a proposed implementation of the assignment layout discussed in #67 .

Resolves #67 
I meant to ask what you had in mind for testing the demo and whether or not the teardown runs, in addition to the unittests that we have for ok.py.