Add missing #imports for header files

Hi, I was trying go get NNNetwork compiled and was running into some trouble.  It appears that there were some missing #imports, such that Xcode was claiming to not know of the existence of various methods (mostly categories), etc.  Anyway here are the changes I needed to make to get it to compile.