Yeah, I wanted to implement a grace period to jumping when falling but ran out of time. Part of the problem is the players hitbox is circular, so it "falls" a little earlier than you think, so like a 3 frame grace period would be perfect.
Thanks for checking it out and the feedback!