Remove unnecessary / confusing code in example