diff --git a/python-api-src/library.cpp b/python-api-src/library.cpp
index 2f3a8e37d8d560ca7e7993ec1f6bc4390beec0b7..7f54cb213c2bc2c2a714b3ad8631441f8c54fa84 100644
--- a/python-api-src/library.cpp
+++ b/python-api-src/library.cpp
@@ -101,4 +101,18 @@ PYBIND11_MODULE(library, m)
 
     m.def("create_participants", &sc2::CreateParticipant, "Create participant from bot", "race"_a, "bot"_a);
     m.def("create_computer", &sc2::CreateComputer, "Create participant from built-in Starcraft computer", "race"_a, "difficulty"_a);
+
+    py::class_<BuildDescription>(m, "BuildDescription")
+        .def(py::init())
+        .def_readwrite("producer_type", &BuildDescription::producer_type)
+        .def_readwrite("result_type", &BuildDescription::result_type)
+        .def_readwrite("ability_used", &BuildDescription::ability_used)
+        .def_readwrite("time", &BuildDescription::time)
+        .def_readwrite("buildings_needed", &BuildDescription::buildings_needed)
+        .def_readwrite("addons_needed", &BuildDescription::addons_needed);
+
+    py::class_<TechTreeImproved>(m, "TechTreeImproved")
+        .def(py::init())
+        .def("load_data", &TechTreeImproved::LoadData)
+        .def("how_to_build", &TechTreeImproved::HowToBuild);
 }