Updated the test a bit to remove standart prints