Small tweaks in README.
PEP8 and code cleanup.