Fix imports in example code