Added missing imports to example code in README

Thanks to Shafique Jamal for suggesting these improvements!