NN: new structure of datasets
NN: new solution of assignment 1