Add documentation, improve demo, fix method name