Removed required scikits learn import