add the new trainning toolkit