fix demo to reflect changes with README