Refine the toy implementation (Results still different with R)