added code to make interfacing with sklearn classifier easy