modified to include connect-flash