Improved the test example