Fixed docstring and import. Added warning to README.