Create krizhevsky_data.py