Ah yes. I'm very likely expanding this and thinking about making it a puzzle platformer. Each level would have a few "sublevels" that you change between (like the current switching level mechanic). I would then make the switching of sublevels into an action in the game and not in the UI.