equation 2 solved in the notebook needs check,    code updated