Start using flash_message helper and partial