Updated flash logic in template and added basic styling