Added test case and working scikit-learn base thing