Machine learning

Might be nice to start these .py files with a comment explaining what's in the file?