Work on forward pass and backward pass