Add more functions and higher-level API
Add example notebook