Add example code demonstrating how to use the python wrapper