pretty update

Removed debug statements and fixed imports