Clean-up code

[programmerPhysicist]

Code has some warnings in python linters, that need resolving. This is for non-security findings.