Implement rest of neural network