added missing import in readme