Add parameters training

Doesn't work