fixed bugs in load_data method in utils.py