Update readme, pep8 and comment the files