[IMP]imporve code.