Change Train method

add SaveWeights, SaveMatrix
small changes