Tidy up plot scripts