Pain during urination with bleeding at end suggests infection of urinary tract( most probably cystitis
). In most of the cases infection subsides after a course of antibiotic, suggested by clinical improvement or normal laboratory report. As pain has subsided it means there is clinical improvement.
I would suggest you to complete the course, give plenty of water to the baby and maintain proper hygiene by regular cleaning of genitalia.