minor update
removing unneeded imports