Add helper to retrieve flow color