Fixed imports and removed debug output