Update gkp_flow_stats.py