Made code review. a little pep8