Add specific visualisation for each problem. Create a function with an optimal control solution as argument.