Remove all print and fix PEP problem