Clean demo notebook formatting

[ci skip]