Add missing import to README example