Removing unneccessary import class