Add better test output