Add au_roc function and cleanup