Add message output to test