PEP8 fixes

Fixes a number of PEP8 violations in affine matrix example.