train changes

Signed-off-by: paramarco <paramarco@gmail.com>