to do implement backprop test on other datasets implement other kinds of output node vectorize as much code as possible