Removing reference to nn lib