Add math to the demo

[aimalz]

The demo goes over use of the code without any connection to the math. In preparation for publication, I'm going to make the demo representative of the paper's content.