[Linh] update code