Add spec for flash message. Improve message as well.