Add some comments, make possible to re train neural network