He is having acute gastroenteritis and needs specific treatment for it.
For vomiting your medicine is ok. Bit may need some antimicrobial for treating stomach infection.
Next Steps
give plenty of fluids including oral rehydration solution and avoid bottle feeding if you are giving .
Health Tips
general hygiene , use boiled water for drinking. give home made foods like dal, rice, curd, fruits , vegetable soups etc.