PEP8 refactor

Manjši popravki zaradi PEP8