Fixing demo;
Fixing docs