Updated example to work with the current version