Clean up imports in example