Update neural.py to use the new load_data function