Remove unused imports

Import cleanup with the help of pylint, please check carefully