Improve phylogeny handling in python code

[nickjcroucher]

Handling of phylogenies needs to be tested with alternative packages (e.g. treeswift).