Modified flash listener to allow custom success messages