updated to work with new flow / heat maps