the code is pretty good now. thanks. @junmin-zhu