Made example code more consistent with the tutorial