[REF] Removing print and unused import