Remove prints
Remove import of dbg
Add TODO