Change demo to save properly