Add pre-training!