When will the updated code be pushed

[Yikai-Wang]

I am interested in your work. Since you said in your readme that the code will be updated, I wonder which part of the current code needs to be updated and when will the updated code to be pushed? Thanks!