Improved the example script