A little code revision to Xiong's work
Minor code revision