removing unneeded imports
fix pylint issues