python script with plot class to generate roc curves