Add missing imports to documentation examples