some more changes to fix bug to simple predict outcome pipeline