made code better by using pylint :cheers: