Temporarely added the pickle files, we should make the code more decopuled