Simplify example code in comment