Cleaned up. Added cohort size 'heat map'