WIP: added decision_function