Got the MSE for fwrd and backward.
Minor fixes.
Added comments