Reorganise functions in traceplotting.py

[Jean-Du]

There are multiple functions on plotting the same type of graph. This should be reorganised to make the code clearer and avoid duplicates.