Some pep8 improvements.
Minor changes