Fixing a couple spots in code example