Get the example working a little better