Tidy up
PEP8