Updating missing imports for doc samples