Add transform_density and branch_density