Styled the flash message and added a helper method