pep8 stylish
make code more pythonic