Add to the improve test
Update the formatting for the attachment