Trying to fix boundary conditions and lowering RMSE for the model