Improve code sample