Updated the example to work with new interfaces

Changing the interface makes the example break. Just a quick fix to make it work so that people can get started with the example.