Update README.
Remove unused import.
Added some docstrings.