small changes to nn