remove print & add more pep8