small changes to the example