iam module: tidy up to meet PEP8 better

Only PEP8 fixes :)