Added optimisation state diagram