polish some code