additional changes to the example code