Updated demo to the new code.
Updated README.