improve code example