[IMP]improved code