Added info.
Added figure and updated the demo code.