Changed filename for pre-train and added ability to fill data