Updated Training tool

by Bauhem