small changes for demo