PEP8 tidy up

A quick PEP8 tidy.