Made the example easier to run