tidy up flash message code