ex.4 done, pep8