applied changes to the demo after new version 1.2.1