update demo to remove usage of other elements in the demo