added missing import statement to code sample in README