Add missing import to IPy demo notebook