Added missing import to README example