Code for some ML workflows