Update so the code follow pep257