Added 2 new options for leaf & branch classes

This PR should fix #46