Add visualization classes for tps cost and joint tps