some cleanup work on the comments of the NN code