PEP8 update

Increase readibility of code