some TODOs and the train method of NN