Update documentation and demo
v0.2.1