Improve code formatting in order to pass pylint.