Make controller responsible for displaying flash