working on forward pass.