extended model to use branch-specific population sizes