Rewriting tests with new IO interface
Rewriting tests with new IO interface