made code edits during demonstration