What is the treatment for constipation in babies?

If your baby is 6 months and under, what I did to my then 3month old son was to add gripe water into his milk during feeding. It helps and he will poop the next day.