Updated demo file with better usage examples