Mode change
Specify the functions to export
Draft on W&B