Optimization checks

It's not quite perfect, e.g. the FE doesn't seem to keep the original allocation when plotting, and it doesn't plot all the scenarios. But it's not too bad :)