More tidying of tpms_plotting.py.