Added MockIO
Converted tests to MockIO