Remove debug code
Add missing import
Remove debug print