[IMP]Improved code
[IMP]improved code