Improve README.md and remove TODO list from .py