removing print statement
PEP8
unused import