Add missing import in README

fixes #6