Fix imports in readme example