Removing unneeded imports