You can buy a probiotic especially for breastfed babies. Like Kelly said, most babies do not need it but given what he has gone through some help would be useful. Go to a health food shop or a naturopath and buy a good quality one, they are usually stored in a fridge if they are 'good'.

Lovely to hear he is on the mend. Take care.