Fixing couple of PEP8 errors.

This is why we are here :)