Update demo code and docs.