neural network code complete but not tested