I need tips or patterns for baby socks that have amazing fit.

Rye socks pattern is amazing and fun to knit and I will make for me and my husband but for Logan, they don't stay on! He wiggles his feet a lot and it just comes off. Then they go into his mouth. :D

Now that he started crawling and pulling himself up to standing position and tries to mimic walking (one leg up and down, just one leg), footwear has become my interest. I hear that soft-sole shoes are better but before I take him shopping, I want to see if I can get away with just socks for now. The thought of buying shoes for a baby who doesn't walk yet feels a little weird right now.