slight changes related to additional example