added printout when (re)training