Interesting take on the theme - I definitely think there's potential there, but I think it needs to make more puzzles out of the elements. Just having to switch to do things feels busy rather than a challenge; perhaps if they were on a cooldown timer or limited uses so you have to plan ahead?