Update assembly_stats.py