[IMP]improve code.