Got this to work thanks to the comments below!
A bit rough around the edges but I like the idea you went for, also making the player choose between which upgrade to get.
More games could benefit from a "lava avoidance system" :)
