Update readme
Fix pep8