Added in the flash message code