Provide complete code for the simple demo.

Missing an import may lead to compilation errors, complete code clarifies what needs to be included.