Removed few unnecesary import and some part of code.