Fix missing import. Tiny tweaks to the README.