Updated documentation and demo