Improve comments and context of the example