Improved code example