Updating prediction section of statsModels.R