Only adding some more prints to some functions in transport.py