[IMP] improve code.