added tests for forward propagation