fix: updated to pass pep8