Updated message type for forward pass and backward pass.