Feature/pep8

few pep8 fixes