Fixed the tests for the console output and input