I too faced the same issue with my baby. I used to give him Aptamil gold formula milk. It has all the essential nutrients which improve digestion. Also, it's stage 1 has prebiotics and HMO, these are helpful in improving gut health too.