Fixing more pep 8