working on the final NN paper