make roc plot script more versatile