make treegen more modeular to we can specify what branch we want generated