making transport plot script more flexible