Much nicer, but only works for one scenario atm