added missing imports in sample code