Creates test_code.py to experiment with scikit learn