Moved dataset loading into ml_lib/parse.py