tweaks
add some training stuffs