Thank you for your query,
You continue the medication as per your doctor advise.
Don't give azithromycin
to your 2 months old baby without a doctor prescription and its safety is not established in infants below the age of 6 months.
Some times difficulty in breathing during breastfeeding
is due to nose block which can be relieved by putting one or two drops of saline nasal drops into each nostril
and clearing them before feeding.
For pain abdomen you can give colic
aid drops as prescribed.
Hope I have answered your query.