add the optimization plots