Added python 2 compatibility and improved README.md