try to make the code importable

Improved README.md