Update simple example to new API