Removes unused imports
Removes some unnecesary code from NN Regression