gotta shorten up your functions bro
add maximize fn using naive bayes