Minor fixes after running pylint
Formatting