Refactor example to be a bit simpler