Tidy up python