I'm curious what you found janky/sloppy about the mechanics so I can improve on them in a post-jam update.
The rotation of the pieces was awkward, a slight misinput that makes the piece collide with ground would make the piece spin and hard to control. This could be fixed by giving the player better control of the spin or easing the spin into a halt very quickly after removed from collision.
Dropping a key-piece and it infinitely falling through the level could be fixed by just giving them a static position to come back to if they fall. Maybe a piece inventory UI?
That’s all, I hope to see an update to this!