made code more inline with PEP8 using pylint