Improve the example a bit