add labels to flows in render.py