Add option for partial credit model