Updated widget's codebase for demo page.