plot_path_tools.py (2 lines): - line 122: # TODO: Needs to be changed to be able to handle different kind of loss - line 432: # TODO: Needs to be changed to be able to handle different kind of loss