Fixing up example code