Update demo with source code