Improved demos and updated documentation