Fix PEP8

I will rebase after tests PR will be merged