Revise example code in comments