We can drop a bunch of code here by using tskit 1.0 features: 1. Use the numba support 2. Use the inherited/derived state stuff. We would need to pin to tskit > 1 then