[IMP] Better the code with pylint