Added code for flash message
Added header