add comment & fix pep8 bug