Remove print and move import statement