Better way to use NN