Improve script.py documentation a bit

Mainly to help myself keep everything straight in my head
Update code to work with sklearn 0.14.1

Fixes an error when running random forest and also a DeprecationWarning.
Output files are now correctly closed.
Add test data and update README with instructions