Slight changes to NN