I like where you're headed. As you said, there is some work to be done polish wise. But jam games are what they are haha time is the ultimate enemy of us all.
One suggestion, fwiw, would be to maybe tweak the timing on the puzzle gameplay to be a little faster/more free flowing. I felt like I was spending a LOT of time watching that ball animation, and waiting to be able to move again.
Good Game.