update Usage of Train
use argv instead of hard-coded path
reduce size of last layer of mlp from 2 to 1
fit to output layer size of mlp