Improved example, updated documentation