Fixed missing imports in example