Add the newly updated training toolkit