removed unused import and print statement