Small fixes
Revert one change, add print