Path to data = argument
Switch to scikit-learn