[IMP]: Remove changes in demo file