Additional pretrain methods