Optimizing imports
Separe code from data and experiments