add in label prediction