Some fixes for the example code in the doc.
Merge pull request #3 from datariot/master

Some fixes for the example code in the doc.