Modified code for Estimation Node